# -*- coding: utf-8 -*-
# @Time    : 2024/2/25
# @Author  : Yangyali
#


import mysql.connector


def connect_to_mysql(host, user, password, database):
    """
    连接到MySQL数据库

    参数:
    host (str): MySQL服务器的主机名或IP地址
    user (str): 用于连接到数据库的用户名
    password (str): 用户的密码
    database (str): 要连接的数据库名

    返回:
    mysql.connector.connection: 如果连接成功，则返回一个连接对象；否则返回None
    """
    try:
        # 创建连接
        connection = mysql.connector.connect(
            host=host,
            user=user,
            password=password,
            database=database
        )
        print("连接成功")
        return connection
    except mysql.connector.Error as err:
        print(f"错误: {err}")
        return None

    # 使用函数
def execute_sql(sql):
    connection = connect_to_mysql(HOST, USER, PASSWORD, DATABASE)
    if connection:
        # 使用连接执行一些操作，例如查询
        cursor = connection.cursor()
        cursor.execute(sql)
        rows = cursor.fetchall()
        # for row in rows:
        #     print(row)

            # 记得在完成操作后关闭连接
        cursor.close()
        connection.close()
        return row


if __name__ == "__main__":
    HOST = "127.0.0.1"  # 你可以更改为你的MySQL服务器地址
    USER = "root"  # 你的MySQL用户名
    PASSWORD = "admin"  # 你的MySQL密码
    DATABASE = "bwapp"  # 你要连接的数据库名

    connection = connect_to_mysql(HOST, USER, PASSWORD, DATABASE)

    if connection:
        # 使用连接执行一些操作，例如查询
        cursor = connection.cursor()
        cursor.execute("SELECT * FROM heroes")
        rows = cursor.fetchall()
        for row in rows:
            print(row)

            # 记得在完成操作后关闭连接
        cursor.close()
        connection.close()
